package com.hunterai.comfyui;

import java.util.List;

import lombok.AllArgsConstructor;
import lombok.Data;

@Data
@AllArgsConstructor
public class TaskStatusResponse {
    private String status;
    private boolean completed;
    private List<ImageInfo> images;
    private String error;

    public static TaskStatusResponse success(List<ImageInfo> images) {
        return new TaskStatusResponse("success", true, images, null);
    }

    public static TaskStatusResponse processing() {
        return new TaskStatusResponse("processing", false, null, null);
    }

    public static TaskStatusResponse error(String message) {
        return new TaskStatusResponse("error", false, null, message);
    }

    public static TaskStatusResponse notFound(String message) {
        return new TaskStatusResponse("not_found", false, null, message);
    }
}